package polarity;

import java.util.TreeMap;
import java.util.Map.Entry;

public class Polarity 
{
	private TreeMap<String, Float> bayes = new TreeMap<String, Float>();
	
	public TreeMap<String, Float> getBayes() { return bayes; }
	
	private int numberOfTweet = 0;
	
	public int getNumberOfTweet() { return numberOfTweet; }
	public void incNumberOfTweet() { numberOfTweet++; }
	
	private TreeMap<String, Integer> vocabulary = new TreeMap<String, Integer>(); // Mots Occurence
	
	public TreeMap<String, Integer> getVocabulary() { return vocabulary; } 
	
	public int getNumberOfWords()
	{
		int number = 0;
		for(Entry<String, Integer> set : vocabulary.entrySet() )
		{
			number += set.getValue();
		}
		return number;
	}
	
}
